################### seq2seq-units
seq2seq-units-train:
	CUDA_VISIBLE_DEVICES=8 python -m tasks.ClassSummary.models.Seq2seq.main \
	--save_path ./model/seq2seq-units/ \
	--input_flag units \
	--train ./tasks/ClassSummary/resource/train.token.jsonl \
	--val ./tasks/ClassSummary/resource/valid.token.jsonl \
	--code_field ./tasks/ClassSummary/resource/code.field \
	--nl_field ./tasks/ClassSummary/resource/nl.field \
	--patience 4 \
	--batch_size 64 \
	--max_epoch 100 \
	--lr 0.001 \
	--max_grad 5 \
	--gpu_num 1 \
	--seed 1024 \
	--check_freq 1

seq2seq-units-hybrid-test:
	CUDA_VISIBLE_DEVICES=5 python -m tasks.ClassSummary.models.Seq2seq.test \
	--input_flag units \
	--test ./tasks/ClassSummary/resource/test.token.jsonl \
	--code_field ./tasks/ClassSummary/resource/code.field \
	--nl_field ./tasks/ClassSummary/resource/nl.field \
	--check_point /home/hlcs/lmc/clsum/model/seq2seq-units/Seq2seq-units-epoch=26-bleu=10.1048.ckpt \
	--save_path ./translate/seq2seq-units-hybrid.jsonl \
	--gpu_num 1 \
	--batch_size 64 \
	--seed 1024

seq2seq-units-cross-test:
	CUDA_VISIBLE_DEVICES=5 python -m tasks.ClassSummary.models.Seq2seq.test \
	--input_flag units \
	--test ./tasks/ClassSummary/resource/cross.token.jsonl \
	--code_field ./tasks/ClassSummary/resource/code.field \
	--nl_field ./tasks/ClassSummary/resource/nl.field \
	--check_point /home/hlcs/lmc/clsum/model/seq2seq-units/Seq2seq-units-epoch=26-bleu=10.1048.ckpt \
	--save_path ./translate/seq2seq-units-cross.jsonl \
	--gpu_num 1 \
	--batch_size 64 \
	--seed 1024

